# -*- coding:utf-8 -*-

# @Time    : 2023/10/18 17:13
# @Author  : jinyuhao
# @Email   : jinyuhao@lingxi.ai
# @Software: LLM_internal



# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #

from agent.llm_agent import LLMAgent

default_template = """
你作为一名用户接到了一个电话，需要你模仿"参考会话记录"中用户的说话方式和对方进行沟通。
请注意以下几点：
1.用参考会话记录中用户的原话和对方沟通，不要自己发挥。
2.在参考会话记录中，如果用户连续说多句话，忽略掉无含义的信息，将原话拼接成一句话来回复。
3.回复的内容为一句话，请给对方回复的空间。

参考会话记录如下：
===
{conversation_template}
===

当前的对话记录如下：
===
{conversation_history}
===
用户：
"""

class GenerateTestSet(LLMAgent):
    """用户问题及疑议判断机器人"""
    def __init__(self, conversation_template, conversation_history):

        self.prompt = default_template.format(conversation_template=conversation_template, conversation_history=conversation_history)
        super().__init__(self.prompt)

if __name__ == '__main__':
    test = """"你这是哪里的？
    哦，不需要了哈。
    我也快五十了。
    姓名啊。
    嗯，
    我听不明白你在说什么。
    嗯，我我最近没时间不去了解哈。
    谢谢。
    哦，零过去过去参观是没有时间的啊。
    这个听了没过去，了解也没用啊，听了一次忘记了。
    谢谢谢谢，嗯，拜拜。"""
    splitlines = test.splitlines()
    for string in splitlines:
        print(string)